leetcode 97 problem solution in python